package com.atguigu.lease.web.admin.service.impl;

import com.atguigu.lease.common.mybatisplus.MinioProperties;
import com.atguigu.lease.web.admin.service.FileService;
import io.minio.*;
import io.minio.errors.*;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

@Service
public class FileServiceImpl implements FileService {
    private final MinioClient minioClient;

    private final MinioProperties minioProperties;

    public FileServiceImpl(MinioClient minioClient, MinioProperties minioProperties) {
        this.minioClient = minioClient;
        this.minioProperties = minioProperties;
    }

    @Override
    public String uploadFile(MultipartFile file) throws ServerException, InsufficientDataException,
            ErrorResponseException, IOException, NoSuchAlgorithmException, InvalidKeyException,
            InvalidResponseException, XmlParserException, InternalException {
        //判断桶是否存在
        boolean exists = minioClient.bucketExists(BucketExistsArgs.builder().
                bucket(minioProperties.getBucketName()).build());
        if (!exists){
            //创建桶
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(minioProperties.getBucketName()).build());
            //设置权限
            String policy = """
                           {
                             "Statement" : [ {
                               "Action" : "s3:GetObject",
                               "Effect" : "Allow",
                               "Principal" : "*",
                               "Resource" : "arn:aws:s3:::%s/*"
                             } ],
                             "Version" : "2012-10-17"
                           }""".formatted(minioProperties.getBucketName());
            minioClient.setBucketPolicy(SetBucketPolicyArgs.builder().
                    bucket(minioProperties.getBucketName())
                    .config(policy).build());
        }
        //上传文件

        //处理文件名重复问题
        String objectName = new SimpleDateFormat("yyyyMMdd").format(new Date()) + "/" +
                UUID.randomUUID().toString().replaceAll("-","") + "_" + file.getOriginalFilename();


        minioClient.putObject(PutObjectArgs.builder().bucket(minioProperties.getBucketName())
                .object(file.getOriginalFilename())
                .stream(file.getInputStream(), file.getSize(), -1).contentType(file.getContentType()).build());

        return String.join("/",minioProperties.getEndpoint(),minioProperties.getBucketName(),objectName);
    }
}
